package _binary_tree

import common.TreeNode
import org.junit.Assert
import org.junit.Test

/**
 *
 * https://leetcode.cn/problems/count-good-nodes-in-binary-tree/description
 *
 * 题型：二叉树的前序遍历
 *
 * ```
 * 1448. 统计二叉树中好节点的数目
 * 给你一棵根为 root 的二叉树，请你返回二叉树中好节点的数目。
 * 「好节点」X 定义为：从根到该节点 X 所经过的节点中，没有任何节点的值大于 X 的值。

 * 示例 1：
 * 输入：root = [3,1,4,3,null,1,5]
 * 输出：4
 * 解释：图中蓝色节点为好节点。
 * 根节点 (3) 永远是个好节点。
 * 节点 4 -> (3,4) 是路径中的最大值。
 * 节点 5 -> (3,4,5) 是路径中的最大值。
 * 节点 3 -> (3,1,3) 是路径中的最大值。
 *
 * 示例 2：
 * 输入：root = [3,3,null,4,2]
 * 输出：3
 * 解释：节点 2 -> (3, 3, 2) 不是好节点，因为 "3" 比它大。
 *
 * 示例 3：
 * 输入：root = [1]
 * 输出：1
 * 解释：根节点是好节点。
 *
 *
 * 提示：
 * 二叉树中节点数目范围是 [1, 10^5] 。
 * 每个节点权值的范围是 [-10^4, 10^4] 。
 * ```
 *
 */
class leetcode_1448 {
    @Test
    fun test_2() {
        val n2 = TreeNode(2)
        val n3 = TreeNode(3)
        val n3_2 = TreeNode(3)
        val n4 = TreeNode(4)

        val root = n3
        n3.left = n3_2
        n3_2.left = n4
        n3_2.right = n2

        val actual = goodNodes(root)
        val expected = 3
        Assert.assertEquals(expected, actual)
    }

    private fun goodNodes(root: TreeNode?): Int {
        /**
        题型：二叉树的前序遍历

        思路：
        符合条件的个数 = 当前节点个数 + 左孩子个数 + 右孩子个数

        空间复杂度：O（N）
        时间复杂度：O（N）

         */
        return collect(root, Int.MIN_VALUE)
    }

    private fun collect(root: TreeNode?, maxValue: Int): Int {
        // 1 确认函数的参数以及返回值
        // 2 确认终止条件
        if (null == root) {
            return 0
        }
        // 3 确认单层递归的处理逻辑:前序遍历
        var count: Int = 0
        // 中
        if (root.`val` >= maxValue) {
            count = count + 1
        }

        val newMaxValue = Math.max(maxValue, root.`val`)

        // 左
        count = count + collect(root.left, newMaxValue)

        // 右
        count = count + collect(root.right, newMaxValue)
        return count
    }
}